from langchain_core.tools import tool
from pydantic import BaseModel, Field


class Calculator(BaseModel):
    a: int = Field(description="第一个数")
    b: int = Field(description="第二个数")


@tool(description="计算两个数的乘法", args_schema=Calculator)
def multi(a: int, b: int) -> int:
    return a * b

print(multi.name)
print(multi.description)
print(multi.args)